This notebook has been launched just for you. It provides an easy way to try out Stan.
**WARNING**
Don't rely on this server for anything you want to last - your notebook will be *deleted after 10 minutes of inactivity*.
In [2]:
%matplotlib inline
import pystan
In [3]:
schools_code = """
data {
int<lower=0> J; // number of schools
real y[J]; // estimated treatment effects
real<lower=0> sigma[J]; // s.e. of effect estimates
}
parameters {
real mu;
real<lower=0> tau;
real eta[J];
}
transformed parameters {
real theta[J];
for (j in 1:J)
theta[j] <- mu + tau * eta[j];
}
model {
eta ~ normal(0, 1);
y ~ normal(theta, sigma);
}
"""
schools_dat = {'J': 8,
'y': [28, 8, -3, 7, -1, 1, 18, 12],
'sigma': [15, 10, 16, 11, 9, 11, 10, 18]}
In [4]:
fit = pystan.stan(model_code=schools_code, data=schools_dat, iter=1000, chains=4)
In [5]:
print(fit)
In [6]:
fit.plot()
Out[6]:
In [ ]: